import json


data_pth = "/home/mbk/rubbish/sen_cls/data/ASAP_ASPECT/trian.tsv"
with open(data_pth, 'r') as f:
    lines = f.read().split('\n')[1:1500]

all_data = []

for line in lines:
    text_a, cate, label = tuple(line.split('\t'))
    prompt = f"""
请完成一个评价对象级情感分类任务。
对象是：{cate},
评价是：{text_a},
请输出 正面 或 负面
注意输出格式是一个词（正面 或 负面），不要输出其它
"""
    temp = {
        "conversations": [
            {
                "value": prompt
            },
            {
                "value": "正面" if label == "1" else "负面"
            }
        ]
    }
    all_data.append(temp)

with open('data1.json', 'w') as f:
    json.dump(all_data, f, ensure_ascii=False, indent=4)
 